from nlp_tools.tasks.classification.abc_model import ABCTaskModel
import json
model_save_path = r'contraband/tf_models/train_model_f1_random/'
loaded_model = ABCTaskModel.load_model(model_save_path)





text = input("请输入：")
while text:
    print(loaded_model.predict([text]))
    text = input("请输入：")